package leetcode.alg.demo;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。
 *
 * 示例：
 * 给定 nums = [2, 7, 11, 15], target = 9
 * 因为 nums[0] + nums[1] = 2 + 7 = 9
 * 所以返回 [0, 1]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/two-sum
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Two_sum {
    public static void main(String[] args){
        int[] array = {2,4,5,8};
        System.out.println(getIndex(array, 9)[0]);
        System.out.println(getIndex(array, 9)[1]);
    }

    public static Integer[] getIndex(int[] nums, int target){
        Set set = new HashSet<Integer>();
        Map map = new HashMap<Integer, Integer>();
        Integer[] results = new Integer[2];
        for(int i = 0; i < nums.length; i ++){
            set.add(nums[i]);
        }
        for(int i = 0; i < nums.length; i ++){
            if(set.contains(target - nums[i])){
                results[0] = i;
                results[1] = (Integer) map.get(target - nums[i]);
            }
            map.put(nums[i], i);
        }
        return results;
    }
}
